package com.example.demo.mqtt;

import com.example.demo.mqtt.AppProviderCallBack;
import lombok.extern.slf4j.Slf4j;
import org.eclipse.paho.client.mqttv3.*;
import org.eclipse.paho.client.mqttv3.persist.MemoryPersistence;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.UUID;

/**
 * @Author：Chan
 * @Package：com.zkkz.boot.mqtt.config
 * @Project：zkkz-boot
 * @name：MqttToAllHelper
 * @Date：2023/9/13 13:51
 * @Describe：APP发消息给用户则通过该工具类进行操作
 */

@Slf4j
@Component
public class AppProviderHelper {
    @Value("${device.username:MQTT1}")
    private String username;

//    @Value("${device.password:public}")
    @Value("${device.password:123456}")
    private String password;

    @Value("${device.url:tcp://127.0.0.1:1883}")
    private String hostUrl;

    @Value("${device.app.providerid:app_provider1}")
    private String clientId;

    //遗嘱对应的topic
    @Value("${device.app.will.topic:app_will_topic}")
    private String defaultTopic;


    /**
     * 客户端对象(服务器作为一个客户端)
     */
    private MqttClient client;
    private MqttConnectOptions options;

    @PostConstruct
    public void connect() {
        try {
            //创建MQTT客户端对象
            client = new MqttClient(hostUrl, clientId+ UUID.randomUUID(), new MemoryPersistence());
            //连接设置
            options = new MqttConnectOptions();
            //是否清空session，设置为false表示服务器会保留客户端的连接记录（订阅主题，qos），客户端重连之后能获取到服务器在客户端断开连接期间推送的消息
            //设置为true表示每次连接到服务端都是以新的身份
            options.setCleanSession(false);
            //设置连接用户名
            options.setUserName(username);
            //设置连接密码
            options.setPassword(password.toCharArray());
            //设置超时时间，单位为秒
            options.setConnectionTimeout(100);
            //设置心跳时间 单位为秒，表示服务器每隔1.5*20秒的时间向客户端发送心跳判断客户端是否在线
            options.setKeepAliveInterval(20);
            //设置遗嘱消息的话题，若客户端和服务器之间的连接意外断开，服务器将发布客户端的遗嘱信息
            options.setWill(defaultTopic, (clientId + "与服务器断开连接").getBytes(), 0, false);
            //设置回调
            client.setCallback(new AppProviderCallBack(client, options));
            client.connect(options);
            client.subscribe("wisdom/hq/+/Snap", 2);//订阅消息主题中的+号，如果不转义就是通配符，表示订阅所有以wisdom/hq/开头，Snap结尾的主题
        } catch (MqttException e) {
            e.printStackTrace();
        }
    }

    /**
     * 发布消息
     * @param topic
     * @param qos  内部消息统一用qos=2；外部消息根据具体情况使用qos的值，目前闸机使用qos=1  内部消息格式按返回来决定
     * @param retained  默认true
     * @param message 内部消息格式按返回来决定;外部消息则需要看接口文档
     *
     * @return
     */
    public boolean publish(String topic, int qos, boolean retained, String message) {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(qos);
        mqttMessage.setRetained(retained);
        mqttMessage.setPayload(message.getBytes());
        //主题目的地，用于发布/订阅消息
        MqttTopic mqttTopic = client.getTopic(topic);
        //提供一种机制来跟踪消息的传递进度。
        //用于在以非阻塞方式（在后台运行）执行发布时跟踪消息的传递进度
        MqttDeliveryToken token;
        try {
            //将指定消息发布到主题，但不等待消息传递完成。返回的token可用于跟踪消息的传递状态。
            //一旦此方法干净地返回，消息就已被客户端接受发布。当连接可用时，将在后台完成消息传递。
            token = mqttTopic.publish(mqttMessage);
            token.waitForCompletion();
        } catch (MqttException e) {
            e.printStackTrace();
            log.info("发送消息过程中出现异常");
            return false;
        }
        return true;
    }
}
